from 卷积神经网络 import train_model

train_model(10, batch_size=16, save_name='test', lr = 0.01, opt = 'Adam')